package hi.storm.test.blot;

import java.util.HashMap;
import java.util.Map;

import org.apache.storm.task.OutputCollector;
import org.apache.storm.task.TopologyContext;
import org.apache.storm.topology.OutputFieldsDeclarer;
import org.apache.storm.topology.base.BaseRichBolt;
import org.apache.storm.tuple.Tuple;

import hi.storm.test.util.StringHelper;

public class SumBolt extends BaseRichBolt {
	private static final long serialVersionUID = 4743224635827696343L;

	/**
	 * 在Bolt启动前执行，提供Bolt启动环境配置的入口 一般对于不可序列化的对象进行实例化。 注:如果是可以序列化的对象，那么最好是使用构造函数。
	 */
	public void prepare(Map<String, Object> map, TopologyContext arg1, OutputCollector collector) {
		System.out.println("prepare:" + map.get("test"));
		this.counts = new HashMap<String, Integer>();
	}

	/**
	 * execute()方法是Bolt实现的核心。 也就是执行方法，每次Bolt从流接收一个订阅的tuple，都会调用这个方法。
	 *
	 */
	public void execute(Tuple tuple) {
		String msg = tuple.getStringByField("count");
		if (StringHelper.isBlank(msg)) {
			return;
		}
		System.out.println("item:" + count + " times");
		/**
		 * 如果不包含该单词，说明在该map是第一次出现 否则进行加1
		 */
		if (!counts.containsKey(msg)) {
			counts.put(msg, 1);
		} else {
			counts.put(msg, counts.get(msg) + 1);
		}
		count++;
	}

	/**
	 * 声明数据格式
	 */
	public void declareOutputFields(OutputFieldsDeclarer arg0) {
	}

	/**
	 * cleanup是IBolt接口中定义,用于释放bolt占用的资源。 Storm在终止一个bolt之前会调用这个方法。
	 */
	@Override
	public void cleanup() {
		System.out.println("===========show word summ============");
		for (Map.Entry<String, Integer> entry : counts.entrySet()) {
			System.out.println(entry.getKey() + ": " + entry.getValue());
		}
		System.out.println("===========finish============");
		System.out.println("Test2Bolt release resource");
	}

	/**
	 * 保存单词和对应的计数
	 */
	private HashMap<String, Integer> counts = null;

	private long count = 1;
}
